Skip to content

Remove find_unused_parameters#21

Open
timmeinhardt wants to merge 1 commit intowjf5203:mainfrom
timmeinhardt:Remove_find_unused_parameters
Open

Remove find_unused_parameters#21
timmeinhardt wants to merge 1 commit intowjf5203:mainfrom
timmeinhardt:Remove_find_unused_parameters

Conversation

@timmeinhardt
Copy link
Copy Markdown

@timmeinhardt timmeinhardt commented Aug 2, 2022

This PR removes the self.output_proj_box from MSDeformAttn if the mode is decode which allows to run torch.nn.parallel.DistributedDataParallel without find_unused_parameters=True. In theory, this should improve training time as the torch backend avoids a forward pass for finding these parameters.

Edit: Just realised that my editor also removed several white spaces in the respective files. I hope this is okay.

@timmeinhardt timmeinhardt changed the title Remove_find_unused_parameters Remove find_unused_parameters Aug 2, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant